package com.h.extensions.config;

import org.springframework.amqp.core.ReturnedMessage;
import org.springframework.amqp.rabbit.connection.ConnectionFactory;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.rabbit.transaction.RabbitTransactionManager;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class RabbitTemplateConfig {

    // 可以解决 设置 confirmCallback 回调函数 影响所有使用 RabbitTemplate的方法，都会调用这个回调函数影响正常的使用
    @Bean
    public RabbitTemplate rabbitTemplate(ConnectionFactory connectionFactory) {
        RabbitTemplate rabbitTemplate = new RabbitTemplate(connectionFactory);
        return rabbitTemplate;
    }
    // 不在 ProducerController 这个生产者中进行编写函数，在新的 rabbitTemplate 对象中进行编写回调函数
    // 这样在使用的时候调用这个对象即可使用
    // 这样就可以解决 重复调用接口的时候会出现的错误
    // 如果在 生产者中进行编写的话，在第一次调用的时候不会出现问题，但是当第二次及以后调用的时候就会出现问题
    // 因为第一次对于这个回调函数已经创建成功了，如果第二次进行调用函数就会再次创建，就会出现错误
    @Bean
    public RabbitTemplate confirmRabbitTemplate(ConnectionFactory connectionFactory) {
        RabbitTemplate rabbitTemplate = new RabbitTemplate(connectionFactory);
        // 设置回调方法
        rabbitTemplate.setConfirmCallback(new RabbitTemplate.ConfirmCallback() {
            @Override
            public void confirm(CorrelationData correlationData, boolean ack, String cause) {
                System.out.println("执行了confirm方法");
                if (ack == true) {
                    System.out.printf("接收到消息, 消息ID: %s /n",
                            correlationData == null ? null : correlationData.getId());
                }else {
                    System.out.printf("未接收到消息, 消息ID: %s, cause: %s /n",
                            correlationData == null ? null : correlationData.getId(), cause);
                    //相应的业务处理
                }
            }
        });
        // 消息被退回时, 回调方法 —— return退回模式
        rabbitTemplate.setMandatory(true);
        rabbitTemplate.setReturnsCallback(new RabbitTemplate.ReturnsCallback() {
            @Override
            public void returnedMessage(ReturnedMessage returnedMessage) {
                System.out.println("消息退回: " + returnedMessage);
            }
        });

        return rabbitTemplate;
    }

    // 事务的操作
    @Bean("transRabbitTemplate")
    public RabbitTemplate transRabbitTemplate(ConnectionFactory connectionFactory) {
        RabbitTemplate rabbitTemplate = new RabbitTemplate(connectionFactory);
        rabbitTemplate.setChannelTransacted(true); //开启事务 —— 必须存在
        return rabbitTemplate;
    }
    @Bean
    public RabbitTransactionManager rabbitTransactionManager(ConnectionFactory connectionFactory) {
        return new RabbitTransactionManager(connectionFactory); // 需要创建RabbitMQ的事务管理器
    }
}
